package com.hopu.oa.until.data;

import com.aliyun.oss.ClientException;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.model.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.UUID;

/**
 * @ClassName: AliyunOSSUtil
 * @Description:
 * @Date: 2021/11/29 11:06
 */
@Service
public class AliyunOSSUtil {
    @Autowired
    private ConstantProperties constantProperties;
    private static SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");

    /**
     * 上传
     *
     * @param file
     * @return
     */
    public String upload(File file) {
        System.out.println("=========>OSS文件上传开始：" + file.getName());
        String endpoint = constantProperties.getEndpoint();
        String accessKeyId = constantProperties.getKeyid();
        String accessKeySecret = constantProperties.getKeysecret();
        String bucketName = constantProperties.getBucketname();
        String fileHost = constantProperties.getFilehost();
        System.out.println(endpoint + "endpoint");
        SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd");
        String dateStr = format.format(new Date());

        if (null == file) {
            return null;
        }

        OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
        try {
            //容器不存在，就创建
            if (!ossClient.doesBucketExist(bucketName)) {
                ossClient.createBucket(bucketName);
                CreateBucketRequest createBucketRequest = new CreateBucketRequest(bucketName);
                createBucketRequest.setCannedACL(CannedAccessControlList.PublicRead);
                ossClient.createBucket(createBucketRequest);
            }
            //创建文件路径
            String fileUrl = fileHost + "/" + (dateStr + "/" + UUID.randomUUID().toString().replace("-", "") + "-" + file.getName());
            //上传文件
            PutObjectResult result = ossClient.putObject(new PutObjectRequest(bucketName, fileUrl, file));
            //设置权限 这里是公开读
            ossClient.setBucketAcl(bucketName, CannedAccessControlList.PublicRead);
            if (null != result) {
                System.out.println("==========>OSS文件上传成功,OSS地址：" + fileUrl);
                return fileUrl;
            }
        } catch (OSSException oe) {
            System.out.println(oe.getMessage());
        } catch (ClientException ce) {
            System.out.println(ce.getMessage());
        } finally {
            //关闭
            ossClient.shutdown();
        }
        return null;
    }


    /**
     * 删除
     *
     * @param fileKey
     * @return
     */
    public String deleteBlog(String fileKey) {
        System.out.println("=========>OSS文件删除开始");
        String endpoint = constantProperties.getEndpoint();
        String accessKeyId = constantProperties.getKeyid();
        String accessKeySecret = constantProperties.getKeysecret();
        String bucketName = constantProperties.getBucketname();
        String fileHost = constantProperties.getFilehost();
        try {
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);

            if (!ossClient.doesBucketExist(bucketName)) {
                System.out.println("==============>您的Bucket不存在");
                return "您的Bucket不存在";
            } else {
                System.out.println("==============>开始删除Object");
                ossClient.deleteObject(bucketName, fileKey);
                System.out.println("==============>Object删除成功：" + fileKey);
                return "==============>Object删除成功：" + fileKey;
            }
        } catch (Exception ex) {
            System.out.println("删除Object失败" + ex);
            return "删除Object失败";
        }
    }

    /**
     * 查询文件名列表
     *
     * @param bucketName
     * @return
     */
    public List<String> getObjectList(String bucketName) {
        List<String> listRe = new ArrayList<>();
        String endpoint = constantProperties.getEndpoint();
        String accessKeyId = constantProperties.getKeyid();
        String accessKeySecret = constantProperties.getKeysecret();
        try {
            System.out.println("===========>查询文件名列表");
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            ListObjectsRequest listObjectsRequest = new ListObjectsRequest(bucketName);
            //列出11111目录下今天所有文件
            listObjectsRequest.setPrefix("11111/" + format.format(new Date()) + "/");
            ObjectListing list = ossClient.listObjects(listObjectsRequest);
            for (OSSObjectSummary objectSummary : list.getObjectSummaries()) {
                System.out.println(objectSummary.getKey());
                listRe.add(objectSummary.getKey());
            }
            return listRe;
        } catch (Exception ex) {
            System.out.println("==========>查询列表失败" + ex);
            return new ArrayList<>();
        }
    }


    public void downloadFile(String fileName, String ossKey, HttpServletRequest request, HttpServletResponse response, String content) {
        String endpoint = constantProperties.getEndpoint();
        String accessKeyId = constantProperties.getKeyid();//你的keyid
        String accessKeySecret = constantProperties.getKeysecret();//你的keysecret
        String bucketName = constantProperties.getBucketname();//bucket名称
        try {
            OSSClient ossClient = new OSSClient(endpoint, accessKeyId, accessKeySecret);
            OSSObject ossObject = ossClient.getObject(bucketName, ossKey);
            BufferedInputStream in = new BufferedInputStream(ossObject.getObjectContent());
            BufferedOutputStream out = new BufferedOutputStream(response.getOutputStream());
            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
            response.setContentType(content);
            byte[] car = new byte[1024];
            int L = 0;
            while ((L = in.read(car)) != -1) {
                out.write(car, 0, L);
            }
            if (out != null) {
                out.flush();
                out.close();
            }
            if (in != null) {
                in.close();
            }
            ossClient.shutdown();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
